cat << 'EOF' > src/yololayer.h
#ifndef YOLO_LAYER_H
#define YOLO_LAYER_H

#ifdef __cplusplus
extern "C" {
#endif

void hello_plugin();
void launch_yolo_kernel(float* input, float* output, int size);

#ifdef __cplusplus
}
#endif

#endif // YOLO_LAYER_H
